package main.leetcode.offer.firstround.from03to50;

import main.datastruc.TreeNode;

/**
 * 26.树的子结构
 *
 * <p>输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
 *
 * <p>B是A的子结构， 即 A中有出现和B相同的结构和节点值。
 *
 * <p>示例 1：输入：A = [1,2,3], B = [3,1] 输出：false
 *
 * <p>示例 2：输入：A = [3,4,5,1,2], B = [4,1] 输出：true
 *
 * <p>限制：0 <= 节点个数 <= 10000
 *
 * <p>来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class ex26 {

    /** [1,1,2,3,4] [1,3]执行不通过，但是提交通过 */
    boolean flag = false;

    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if (A == null || B == null) return false;
        return dfs(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }

    public boolean dfs(TreeNode A, TreeNode B) {
        if (B == null) return true;
        if (A == null) return false;
        return A.val == B.val && dfs(A.right, B.right) && dfs(A.left, B.left);
    }

    public boolean isSubStructure1(TreeNode A, TreeNode B) {
        if (B == null) return flag;
        if (A == null) return false;
        if (A.val == B.val) {
            flag = true;
            return isSubStructure(A.left, B.left) && isSubStructure(A.right, B.right);
        } else return !flag && (isSubStructure(A.left, B) || isSubStructure(A.right, B));
    }
}
